home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2009 February / PCWFEB09.iso / Software / Linux / Kubuntu 8.10 / kubuntu-8.10-desktop-i386.iso / casper / filesystem.squashfs / usr / share / python / pyversions.py < prev   
Encoding:
Python Source  |  2008-06-23  |  11.9 KB  |  328 lines
  1. #! /usr/bin/python
  2.  
  3. import os, re, sys
  4. try:
  5.     SetType = set
  6. except NameError:
  7.     import sets
  8.     SetType = sets.Set
  9.     set = sets.Set
  10.  
  11. _defaults = None
  12. def read_default(name=None):
  13.     global _defaults
  14.     from ConfigParser import SafeConfigParser, NoOptionError
  15.     if not _defaults:
  16.         if os.path.exists('/usr/share/python/debian_defaults'):
  17.             config = SafeConfigParser()
  18.             config.readfp(file('/usr/share/python/debian_defaults'))
  19.             _defaults = config
  20.     if _defaults and name:
  21.         try:
  22.             value = _defaults.get('DEFAULT', name)
  23.         except NoOptionError:
  24.             raise ValueError
  25.         return value
  26.     return None
  27.  
  28. def parse_versions(vstring):
  29.     import operator
  30.     operators = { None: operator.eq, '=': operator.eq,
  31.                   '>=': operator.ge, '<=': operator.le,
  32.                   '<<': operator.lt
  33.                   }
  34.     vinfo = {}
  35.     exact_versions = set([])
  36.     version_range = set(supported_versions(version_only=True))
  37.     relop_seen = False
  38.     for field in vstring.split(','):
  39.         field = field.strip()
  40.         if field == 'all':
  41.             vinfo['all'] = 'all'
  42.             continue
  43.         if field in ('current', 'current_ext'):
  44.             vinfo['current'] = field
  45.             continue
  46.         vinfo.setdefault('versions', set())
  47.         ve = re.compile('(>=|<=|<<|=)? *(\d\.\d)$')
  48.         m = ve.match(field)
  49.         try:
  50.             op, v = m.group(1), m.group(2)
  51.             if op in (None, '='):
  52.                 exact_versions.add(v)
  53.             else:
  54.                 relop_seen = True
  55.                 filtop = operators[op]
  56.                 version_range = [av for av in version_range if filtop(av ,v)]
  57.         except Exception:
  58.             raise ValueError, 'error parsing Python-Version attribute'
  59.     if 'versions' in vinfo:
  60.         vinfo['versions'] = exact_versions
  61.         if relop_seen:
  62.             vinfo['versions'] = exact_versions.union(version_range)
  63.     return vinfo
  64.  
  65. _old_versions = None
  66. def old_versions(version_only=False):
  67.     global _old_versions
  68.     if not _old_versions:
  69.         try:
  70.             value = read_default('old-versions')
  71.             _old_versions = [s.strip() for s in value.split(',')]
  72.         except ValueError:
  73.             _old_versions = []
  74.     if version_only:
  75.         return [v[6:] for v in _old_versions]
  76.     else:
  77.         return _old_versions
  78.  
  79. _unsupported_versions = None
  80. def unsupported_versions(version_only=False):
  81.     global _unsupported_versions
  82.     if not _unsupported_versions:
  83.         try:
  84.             value = read_default('unsupported-versions')
  85.             _unsupported_versions = [s.strip() for s in value.split(',')]
  86.         except ValueError:
  87.             _unsupported_versions = []
  88.     if version_only:
  89.         return [v[6:] for v in _unsupported_versions]
  90.     else:
  91.         return _unsupported_versions
  92.  
  93. _supported_versions = None
  94. def supported_versions(version_only=False):
  95.     global _supported_versions
  96.     if not _supported_versions:
  97.         try:
  98.             value = read_default('supported-versions')
  99.             _supported_versions = [s.strip() for s in value.split(',')]
  100.         except ValueError:
  101.             cmd = ['/usr/bin/apt-cache', '--no-all-versions',
  102.                    'show', 'python-all']
  103.             try:
  104.                 import subprocess
  105.                 p = subprocess.Popen(cmd, bufsize=1,
  106.                                      shell=False, stdout=subprocess.PIPE)
  107.                 fd = p.stdout
  108.             except ImportError:
  109.                 fd = os.popen(' '.join(cmd))
  110.             depends = None
  111.             for line in fd:
  112.                 if line.startswith('Depends:'):
  113.                     depends = line.split(':', 1)[1].strip().split(',')
  114.             fd.close()
  115.             depends = [re.sub(r'\s*(\S+)[ (]?.*', r'\1', s) for s in depends]
  116.             _supported_versions = depends
  117.     if version_only:
  118.         return [v[6:] for v in _supported_versions]
  119.     else:
  120.         return _supported_versions
  121.  
  122. _default_version = None
  123. def default_version(version_only=False):
  124.     global _default_version
  125.     if not _default_version:
  126.         _default_version = link = os.readlink('/usr/bin/python')
  127.     # consistency check
  128.     debian_default = read_default('default-version')
  129.     if not _default_version in (debian_default, os.path.join('/usr/bin', debian_default)):
  130.         raise ValueError, "the symlink /usr/bin/python does not point to the python default version. It must be reset to point to %s" % debian_default
  131.     _default_version = debian_default
  132.     if version_only:
  133.         return _default_version[6:]
  134.     else:
  135.         return _default_version
  136.  
  137. def requested_versions(vstring, version_only=False):
  138.     versions = None
  139.     vinfo = parse_versions(vstring)
  140.     supported = supported_versions(version_only=True)
  141.     if len(vinfo) == 1:
  142.         if 'all' in vinfo:
  143.             versions = supported
  144.         elif 'current' in vinfo:
  145.             versions = [default_version(version_only=True)]
  146.         else:
  147.             versions = vinfo['versions'].intersection(supported)
  148.     elif 'all' in vinfo and 'current' in vinfo:
  149.         raise ValueError, "both `current' and `all' in version string"
  150.     elif 'all' in vinfo:
  151.         versions = versions = vinfo['versions'].intersection(supported)
  152.     elif 'current' in vinfo:
  153.         current = default_version(version_only=True)
  154.         if not current in vinfo['versions']:
  155.             raise ValueError, "`current' version not in supported versions"
  156.         versions = [current]
  157.     else:
  158.         raise ValueError, 'error in version string'
  159.     if not versions:
  160.         raise ValueError, 'empty set of versions'
  161.     if version_only:
  162.         return versions
  163.     else:
  164.         return ['python%s' % v for v in versions]
  165.  
  166. def installed_versions(version_only=False):
  167.     import glob
  168.     supported = supported_versions()
  169.     versions = [os.path.basename(s)
  170.                 for s in glob.glob('/usr/bin/python[0-9].[0-9]')
  171.                 if os.path.basename(s) in supported]
  172.     versions.sort()
  173.     if version_only:
  174.         return [v[6:] for v in versions]
  175.     else:
  176.         return versions
  177.  
  178. class ControlFileValueError(ValueError):
  179.     pass
  180. class MissingVersionValueError(ValueError):
  181.     pass
  182.  
  183. def extract_pyversion_attribute(fn, pkg):
  184.     """read the debian/control file, extract the XS-Python-Version
  185.     field; check that XB-Python-Version exists for the package."""
  186.  
  187.     version = None
  188.     sversion = None
  189.     section = None
  190.     for line in file(fn):
  191.         line = line.strip()
  192.         if line == '':
  193.             if pkg == 'Source':
  194.                 break
  195.             section = None
  196.         elif line.startswith('Source:'):
  197.             section = 'Source'
  198.         elif line.startswith('Package: ' + pkg):
  199.             section = pkg
  200.         elif line.startswith('XS-Python-Version:'):
  201.             if section != 'Source':
  202.                 raise ValueError, \
  203.                       'attribute XS-Python-Version not in Source section'
  204.             sversion = line.split(':', 1)[1].strip()
  205.         elif line.startswith('XB-Python-Version:'):
  206.             if section == pkg:
  207.                 version = line.split(':', 1)[1].strip()
  208.     if section == None:
  209.         raise ControlFileValueError, 'not a control file'
  210.     if pkg == 'Source':
  211.         if sversion == None:
  212.             raise MissingVersionValueError, \
  213.                   'missing XS-Python-Version in control file'
  214.         return sversion
  215.     if version == None:
  216.         raise MissingVersionValueError, \
  217.               'missing XB-Python-Version for package `%s' % pkg
  218.     return version
  219.  
  220. # compatibility functions to parse debian/pyversions
  221.  
  222. def version_cmp(ver1,ver2):
  223.     v1=[int(i) for i in ver1.split('.')]
  224.     v2=[int(i) for i in ver2.split('.')]
  225.     return cmp(v1,v2)
  226.  
  227. def requested_versions_bis(vstring, version_only=False):
  228.     versions = []
  229.     py_supported_short = supported_versions(version_only=True)
  230.     for item in vstring.split(','):
  231.         v=item.split('-')
  232.         if len(v)>1:
  233.             if not v[0]:
  234.                 v[0] = py_supported_short[0]
  235.             if not v[1]:
  236.                 v[1] = py_supported_short[-1]
  237.             for ver in py_supported_short:
  238.                 try:
  239.                     if version_cmp(ver,v[0]) >= 0 \
  240.                            and version_cmp(ver,v[1]) <= 0:
  241.                         versions.append(ver)
  242.                 except ValueError:
  243.                     pass
  244.         else:
  245.             if v[0] in py_supported_short:
  246.                 versions.append(v[0])
  247.     versions.sort(version_cmp)
  248.     if not versions:
  249.         raise ValueError, 'empty set of versions'
  250.     if not version_only:
  251.         versions=['python'+i for i in versions]
  252.     return versions
  253.  
  254. def extract_pyversion_attribute_bis(fn):
  255.     vstring = file(fn).readline().rstrip('\n')
  256.     return vstring
  257.  
  258. def main():
  259.     from optparse import OptionParser
  260.     usage = '[-v] [-h] [-d|--default] [-s|--supported] [-i|--installed] [-r|--requested <version string>|<control file>]'
  261.     parser = OptionParser(usage=usage)
  262.     parser.add_option('-d', '--default',
  263.                       help='print the default python version',
  264.                       action='store_true', dest='default')
  265.     parser.add_option('-s', '--supported',
  266.                       help='print the supported python versions',
  267.                       action='store_true', dest='supported')
  268.     parser.add_option('-r', '--requested',
  269.                       help='print the python versions requested by a build; the argument is either the name of a control file or the value of the XS-Python-Version attribute',
  270.                       action='store_true', dest='requested')
  271.     parser.add_option('-i', '--installed',
  272.                       help='print the installed supported python versions',
  273.                       action='store_true', dest='installed')
  274.     parser.add_option('-v', '--version',
  275.                       help='print just the version number(s)',
  276.                       default=False, action='store_true', dest='version_only')
  277.     opts, args = parser.parse_args()
  278.     program = os.path.basename(sys.argv[0])
  279.  
  280.     if opts.default and len(args) == 0:
  281.         try:
  282.             print default_version(opts.version_only)
  283.         except ValueError, msg:
  284.             print "%s:" % program, msg
  285.             sys.exit(1)
  286.     elif opts.supported and len(args) == 0:
  287.         print ' '.join(supported_versions(opts.version_only))
  288.     elif opts.installed and len(args) == 0:
  289.         print ' '.join(installed_versions(opts.version_only))
  290.     elif opts.requested and len(args) <= 1:
  291.         if len(args) == 0:
  292.             versions = 'debian/control'
  293.         else:
  294.             versions = args[0]
  295.         try:
  296.             if os.path.isfile(versions):
  297.                 fn = versions
  298.                 try:
  299.                     vstring = extract_pyversion_attribute(fn, 'Source')
  300.                     vs = requested_versions(vstring, opts.version_only)
  301.                 except ControlFileValueError:
  302.                     sys.stderr.write("%s: not a control file: %s, " \
  303.                                      % (program, fn))
  304.                     sys.exit(1)
  305.                 except MissingVersionValueError:
  306.                     fn = os.path.join(os.path.dirname(fn), 'pyversions')
  307.                     sys.stderr.write("%s: missing XS-Python-Version in control file, fall back to %s\n" \
  308.                                      % (program, fn))
  309.                     try:
  310.                         vstring = extract_pyversion_attribute_bis(fn)
  311.                         vs = requested_versions_bis(vstring, opts.version_only)
  312.                     except IOError:
  313.                         sys.stderr.write("%s: missing debian/pyversions file, fall back to supported versions\n" \
  314.                                          % program)
  315.                         vs = supported_versions(opts.version_only)
  316.             else:
  317.                 vs = requested_versions(versions, opts.version_only)
  318.             print ' '.join(vs)
  319.         except ValueError, msg:
  320.             sys.stderr.write("%s: %s\n" % (program, msg))
  321.             sys.exit(1)
  322.     else:
  323.         sys.stderr.write("usage: %s %s\n" % (program, usage))
  324.         sys.exit(1)
  325.  
  326. if __name__ == '__main__':
  327.     main()
  328.